iT邦幫忙

2024 iThome 鐵人賽

DAY 12
0
AI/ ML & Data

我的深度學習-從0開始實作物件偵測系列 第 12

【Day 12】前置作業 - Pytorch安裝

  • 分享至 

  • xImage
  •  

一、前言

在開始學習深度學習之前,首先必須先把環境設定好,而Pytorch作為深度學習密不可分的項目,因此今天我們要來說說這個東西要怎麼安裝,在安裝之前,要先對自己的電腦有初步的認識,確定自己顯卡的配置才能正確安裝適合自己的版本,而這個環節如果出了問題,就會影響到後面的實作,接著讓我們開始吧!

二、顯卡配置(無Nvidia顯卡的略過)

  1. 可先進入工作管理員中,在GPU的欄列看到使用的顯卡。(筆者使用的是Nvidia的RTX 3070)
    image
  2. 接著在Cmd中輸入指令nvidia-smi可查看目前的顯卡驅動版本,如果版本低於396.26請至官網更新驅動。
    image

三、安裝Pytorch

  1. 請到Pytorch官網的下載頁面
  2. 請根據硬體來選擇下方的圖示,例如筆者使用的是Windows系統,在這邊我選擇使用pip套件進行安裝,語言選擇Python,請確認裝置中是否有Nvidia的顯示卡,才可選擇CUDA版本,無顯卡則選擇使用CPU,使用CUDA可使訓練效率較快。

截圖 2024-07-11 上午11.41.38
3. 接著在cmd中輸入以下指令。

不過pytorch官方並沒有提出使用虛擬環境安裝的方式(也有可能是筆者沒有找到),因此在這邊我們依舊使用pip來安裝pytorch。

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
  1. 接著輸入y安裝套件。

  2. 出現done代表安裝完成

四、測試

可使用簡單的程式測試一下pytorch的功能,在cmd中先輸入python後,接著輸入下列程式。

import torch
x = torch.rand(5, 3)
print(x)

輸出會類似於下圖
截圖 2024-08-12 下午3.38.52

也可使用下列指令確定GPU驅動與CUDA是否啟用成功。

import torch
torch.cuda.is_available()

截圖 2024-08-12 下午3.41.09

出現True就代表啟用成功了,那今天就到這結束,明天見。

image


上一篇
【Day 11】好用套件管理工具分享 - pipenv
下一篇
【Day 13】前置作業 - 初始化環境
系列文
我的深度學習-從0開始實作物件偵測30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言